Fix searching for weights
