Fixed imports.
Fixed missing nose import when running the tests.
TEST: Updated unit tests.