Cleaned imports on test