Make sleep segmentation work with new interface

Signed-off-by: Maxim Osipov <maxim@transclucator>