Make proper implementation of wake method