Bugfix and updated TODO