Cleaned up that test