Fixed problem with testing.