Resolved bug

Signed-off-by: David Havl <info@davidhavl.com>