Fixed casting bug

Signed-off-by: Nick Tolomiczenko <nick.tolomiczenko@gmail.com>