Revert "Calculates correct penalty"

This reverts commit 48e59b5fb117595cd4a42255888f57d7e8eed81b.