improve levels unit test