Fixed issue with testing