1 files changed, 16 insertions, 13 deletions

diff --git a/ttrun/tests/test_ttrun.py b/ttrun/tests/test_ttrun.py
index 06143b7..2156c40 100644
--- a/ttrun/tests/test_ttrun.py
+++ b/ttrun/tests/test_ttrun.py
@@ -1,16 +1,19 @@
-# -*- coding: utf-8 -*-
-
-# Licensed under the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License. You may obtain
-# a copy of the License at
+# Copyright (c) 2016 Red Hat, Inc
+#
+# This file is part of ttrun
+#
+# ttrun is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
 #
-#      http://www.apache.org/licenses/LICENSE-2.0
+# ttrun is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
 #
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
-# License for the specific language governing permissions and limitations
-# under the License.
+# You should have received a copy of the GNU General Public License
+# along with Ansible.  If not, see <http://www.gnu.org/licenses/>.
 
 """
 test_ttrun
@@ -19,10 +22,10 @@ test_ttrun
 Tests for `ttrun` module.
 """
 
-from ttrun.tests import base
+import testtools
 
 
-class TestTtrun(base.TestCase):
+class TestTtrun(testtools.TestCase):
 
     def test_something(self):
         pass