Fixed getWait method signature

Fixed getWait method signature