Bugfix.

Signed-off-by: Viktor Tomkovič <viktor.tomkovic@gmail.com>