Fixes #14

Signed-off-by: oskar94 <oskar.warme@gmail.com>