bugfix from last commit - forgot super __init__