Fix #21. IDK how I could program this so bad...