Fix for Tests,previously forgotten