Changed wafer test function to use it for bare module test