Cleanup, bugfix

Signed-off-by: Dominik Schadow <dominikschadow@googlemail.com>