still getting a bug in tests