Bug fix.

Signed-off-by: bregell <johan_bregell@hotmail.com>