Fix minor error in non-working code ;)

Signed-off-by: Maxim Osipov <maxim.osipov@gmail.com>