Corrected test to work with PS